// 将上面函数的参数改为cin即可